# -*- coding: utf-8 -*-
from si_4k.sistema.models import *
from django.contrib.sites.models import Site
from django import template
from datetime import datetime

register = template.Library()

@register.filter
def checar_unidade(nome_unidade,id_aluno):
    try:
        unidade = Unidade.objects.get(nome=nome_unidade)
        aluno = Aluno.objects.filter(id=id_aluno).get(unidade=unidade)
        return True
    except:
        return False

@register.filter
def checar_unidade2(nome_unidade,id_funcionario):
    try:
        unidade = Unidade.objects.get(nome=nome_unidade)
        detalhe = DetalheUsuario.objects.filter(id=id_funcionario).get(unidade=unidade)
        return True
    except:
        return False